Full Attention Strikes Back: Transferring Full Attention into Sparse within Hundred Training Steps
全注意力反击:百步训练内将全注意力转化为稀疏注意力
🧭 长上下文推理 · 注意力稀疏化
Nanjing University · Alibaba Group
发现全注意力LLM存在内在稀疏性,通过轻量索引和动态选择仅需数百步训练即可达到近乎无损的稀疏推理
📌 工作参考价值(4 点)
- 动态 top-p 筛选: 对电商搜索/推荐中的长序列用户行为编码,借鉴 query-dependent 的动态 token 预算,代替固定 top-k,可保留关键信息同时降低计算开销。
- 检索头识别与保留: 在 Agent 长对话记忆或商品详细页理解中,仅保留少数注意力头进行全长度上下文检索,其余头使用稀疏模式,可大幅压缩 KV 缓存。
- 低维索引器设计: 用 16 维索引矩阵替代完整注意力分数计算,适合实时性要求高的生成式检索或召回阶段,可在预填充阶段大幅提速。
- 少步迁移训练: 利用已有的全注意力模型,仅需数百步微调即可迁移到稀疏推理,无需从头预训练稀疏模型,节省训练资源且保持精度,适合业务快速迭代。